package pkg

import (
	"common/model"
	_ "github.com/mattn/go-sqlite3"
	"log"
	"xorm.io/xorm"
)

var Engine *xorm.Engine

var DbName = "project.db"

func initDB() {
	err := Engine.Sync2(new(model.User))
	if err != nil {
		log.Fatal(err)
	}

	log.Print("初始化表结构成功")
}

func init() {
	var err error
	// 创建数据库引擎
	Engine, err = xorm.NewEngine("sqlite3", DbName)
	if err != nil {
		log.Fatal("数据库连接失败：", err)
	}

	// 开启SQL日志，方便调试
	Engine.ShowSQL(true)

	// 测试连接
	if err := Engine.Ping(); err != nil {
		log.Fatal("数据库连接测试失败：", err)
	}

	log.Print("数据库连接成功！")

	initDB()
}
